ClientMarshallerFactory.java

/**
 *    Copyright 2014 Opower, Inc.
 *    Licensed under the Apache License, Version 2.0 (the "License");
 *    you may not use this file except in compliance with the License.
 *    You may obtain a copy of the License at
 *
 *    http://www.apache.org/licenses/LICENSE-2.0
 *
 *    Unless required by applicable law or agreed to in writing, software
 *    distributed under the License is distributed on an "AS IS" BASIS,
 *    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *    See the License for the specific language governing permissions and
 *    limitations under the License.
 **/
package com.opower.rest.client.generator.marshallers;

import com.opower.rest.client.generator.util.FindAnnotation;
import com.opower.rest.client.generator.util.MediaTypeHelper;

import javax.ws.rs.CookieParam;
import javax.ws.rs.FormParam;
import javax.ws.rs.HeaderParam;
import javax.ws.rs.MatrixParam;
import javax.ws.rs.PathParam;
import javax.ws.rs.QueryParam;
import javax.ws.rs.core.Context;
import javax.ws.rs.core.Cookie;
import javax.ws.rs.core.MediaType;
import java.lang.annotation.Annotation;
import java.lang.reflect.AccessibleObject;
import java.lang.reflect.Method;
import java.lang.reflect.Type;

public class ClientMarshallerFactory {

    public static Marshaller[] createMarshallers(Class declaringClass, Method method) {
        return createMarshallers(declaringClass, method, MediaType.APPLICATION_JSON_TYPE);
    }

    public static Marshaller[] createMarshallers(Class declaringClass, Method method, MediaType defaultConsumes) {
        Marshaller[] params = new Marshaller[method.getParameterTypes().length];
        for (int i = 0; i < method.getParameterTypes().length; i++) {
            Class<?> type = method.getParameterTypes()[i];
            Annotation[] annotations = method.getParameterAnnotations()[i];
            Type genericType = method.getGenericParameterTypes()[i];
            AccessibleObject target = method;
            params[i] = ClientMarshallerFactory.createMarshaller(declaringClass, type, annotations, genericType, target, defaultConsumes, false);
        }
        return params;
    }

    public static Marshaller createMarshaller(Class<?> declaring, Class<?> type,
                                              Annotation[] annotations, Type genericType, AccessibleObject target, MediaType defaultConsumes,
                                              boolean ignoreBody) {
        Marshaller marshaller = null;

        QueryParam query;
        HeaderParam header;
        MatrixParam matrix;
        PathParam uriParam;
        CookieParam cookie;
        FormParam formParam;
        // Form form;

        if ((query = FindAnnotation.findAnnotation(annotations, QueryParam.class)) != null) {
            marshaller = new QueryParamMarshaller(query.value());
        } else if ((header = FindAnnotation.findAnnotation(annotations,
                HeaderParam.class)) != null) {
            marshaller = new HeaderParamMarshaller(header.value());
        } else if ((cookie = FindAnnotation.findAnnotation(annotations,
                CookieParam.class)) != null) {
            marshaller = new CookieParamMarshaller(cookie.value());
        } else if ((uriParam = FindAnnotation.findAnnotation(annotations,
                PathParam.class)) != null) {
            marshaller = new PathParamMarshaller(uriParam.value());
        } else if ((matrix = FindAnnotation.findAnnotation(annotations,
                MatrixParam.class)) != null) {
            marshaller = new MatrixParamMarshaller(matrix.value());
        } else if ((formParam = FindAnnotation.findAnnotation(annotations,
                FormParam.class)) != null) {
            marshaller = new FormParamMarshaller(formParam.value());
        } else if ((FindAnnotation.findAnnotation(annotations,
                Context.class)) != null) {
            marshaller = new NOOPMarshaller();
        } else if (type.equals(Cookie.class)) {
            marshaller = new CookieParamMarshaller(null);
        } else if (!ignoreBody) {
            MediaType mediaType = MediaTypeHelper.getConsumes(declaring, target);
            if (mediaType == null)
                mediaType = defaultConsumes;
            if (mediaType == null) {
                throw new RuntimeException(
                        "You must define a @Consumes type on your client method or interface, or supply a default");
            }
            marshaller = new MessageBodyParameterMarshaller(mediaType, type,
                    genericType, annotations);
        }
        return marshaller;
    }
}